package com.ss.familyserviceplatform.poitest;

/**
 * @Auther: ljy.s
 * @Date: 2022/9/20 - 09 - 20 - 0:14
 */

import org.apache.ibatis.annotations.Param;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.List;

/**
 * 利用Poi读取Excel表格数据
 */
public class ReadDemo {
    public static void main(String[] args) {
        try {
            List<Product> products = read("D:\\Java\\SourceCode\\hejiayun/ss.xlsx");
            for (Product product : products) {
                System.out.println(product);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static List<Product> read(String path) throws Exception {
        List<Product> products = new ArrayList<>();
        // 1.创建输入流
        FileInputStream fip = new FileInputStream(path);
        // 2.在输入流中获取工作簿（excel表）
        XSSFWorkbook workbook = new XSSFWorkbook(fip);
        // 3.在工作簿中获取目标的工作表
        Sheet sheet = workbook.getSheetAt(0);
        // 4.获取工作表中的行数（有数据的）
        int rowNum = sheet.getPhysicalNumberOfRows();
        // 因为第一行数据为标题，不需要，所以要跳过第一行
        for (int i = 1; i < rowNum; i++) {
            // 5.获取除了第一行以外的所有行
            Row row = sheet.getRow(i);
            if (row != null) {
                // 用于保存每条数据的集合
                List<String> list = new ArrayList<>();
                for (Cell cell : row) {
                    if (cell != null) {
                        // 把单元格中所有的数据格式设置为String
                        cell.setCellType(Cell.CELL_TYPE_STRING);
                        // 获取所有单元格数据
                        String value = cell.getStringCellValue();
                        if (value != null && !value.equals("")) {
                            // 将每一个单元格的数据存储到集合中
                            list.add(value);
                        }
                    }
                }
                // 把获取到的每一条数据封装成一个Product类型
                if (list.size() > 0) {
                    Product product = new Product(
                            Integer.parseInt(list.get(0)),
                            list.get(1),
                            Double.parseDouble(list.get(2)),
                            Integer.parseInt(list.get(3)));
                    products.add(product);
                }
            }
        }
        return products;
    }
}
